LineGraph.java

package com.brejza.matt.habmodem;

import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.TreeMap;
import java.util.concurrent.ConcurrentHashMap;

import org.achartengine.ChartFactory;
import org.achartengine.model.XYMultipleSeriesDataset;
import org.achartengine.model.XYSeries;
import org.achartengine.renderer.XYMultipleSeriesRenderer;
import org.achartengine.renderer.XYSeriesRenderer;

import com.brejza.matt.habmodem.Payload;
import ukhas.Telemetry_string;

import android.content.Context;
import android.graphics.Color;
import android.graphics.Paint.Align;
import android.view.View;

public class LineGraph {
	
	private ConcurrentHashMap<String,Payload> _data;
	List<String> listpayloads = new ArrayList<String>();
	List<String> listfields = new ArrayList<String>();

	
	public LineGraph(ConcurrentHashMap<String,Payload> data)
	{
		_data = data;
	}
	
	public void putPayload(String call)
	{
		call = call.toUpperCase();
		listpayloads.add(call);
	}
	
	public void togglePayload(String call)
	{
		call = call.toUpperCase();
		if (listpayloads.contains(call))
			listpayloads.remove(call);
		else
			listpayloads.add(call);
	}
	
	public void addPayload(String call)
	{
		call = call.toUpperCase();
		if (listpayloads.contains(call))
			;
		else
			listpayloads.add(call);
	}
	
	public void clearPayload(String call)
	{
		call = call.toUpperCase();
		if (listpayloads.contains(call))
			listpayloads.remove(call);
		else
			;
	}
	
	public boolean addField(String field)
	{
		int differentFields = 0;
		boolean gotTemp = false;
		/*
		if (listfields.size() < 2){
			listfields.add(field);
			return true;
		}
		else
			return false; */
		
		if (!listfields.contains(field))
		{
			for (int i = 0; i < listfields.size(); i++)
			{
				if (isTemperature(listfields.get(i)))
				{
					if (!gotTemp)
					{
						gotTemp = true;
						differentFields++;
					}
				}
				else
				{
					differentFields++;
				}				
			}
			
			
			if (differentFields < 2 || (gotTemp && differentFields == 2 && isTemperature(field)))
				{
					listfields.add(field);
					return true;
				}
				else
					return false;	
		}
		
		return true; 
	}
		
	private boolean isTemperature(String str)
	{
		return (str.contains("temperature"));
	}
	
	public void clearField(String field)
	{
		if (listfields.contains(field))
		{
			listfields.remove(field);
		}
	}
	
	
	public View getView(Context context)
	{
		long minTime = System.currentTimeMillis() + 60*60*1000;
		long maxTime = 0;
		
		if(listpayloads.size() < 1)
			return null;
		if (Math.min(2, listfields.size()) == 0)
			return null;
		
		int differentFields = 0;
		int tempLoc = -1;
		for (int i = 0; i < listfields.size(); i++){
			if (isTemperature(listfields.get(i))){
				if (tempLoc < 0){
					tempLoc = i;
					differentFields++;
				}
			}
			else
				differentFields++;						
		}
		
		
		
		XYMultipleSeriesDataset dataset = new XYMultipleSeriesDataset();
		XYMultipleSeriesRenderer mrend = new XYMultipleSeriesRenderer( Math.min(2, differentFields));
		int cnt = 0;
		
		
		for (int i = 0; i < listpayloads.size(); i++)
		{
			String call = listpayloads.get(i);
			int colour = _data.get(call).colour;
			
			for (int f = 0; f < listfields.size(); f++)
			{			
				String field = listfields.get(f);
				double m = 1;
				double c = 0;
				
				if (_data.containsKey(call))
				{
					int findex = 0;
					if (_data.get(call).telemetryConfig != null){
						findex = _data.get(call).telemetryConfig.getIndex(field);
						if (findex >= 0){
							m = _data.get(call).telemetryConfig.getFieldScale(findex);
							c = _data.get(call).telemetryConfig.getFieldOffset(findex);
						}
					}
					if (findex >= 0){
						TreeMap<Long,Telemetry_string> sen = _data.get(call).data;
						if (sen.size() > 1)
						{
							
							
							int axis = 0;
							if (!isTemperature(listfields.get(f))){
								if (tempLoc == 0)
									axis = 1;	
								else
									axis = f;
							}
							else
								axis = tempLoc;
							
							XYSeries series = new XYSeries(_data.get(call).callsign + " - " + field,axis);
							
							for (TreeMap.Entry<Long,Telemetry_string> entry : sen.entrySet())
							{				
								if (field == "altitude")
								{
									if (entry.getValue().coords.alt_valid)						
										series.add(entry.getKey(),(int)entry.getValue().coords.altitude);	
								}
								else
								{
									if (entry.getValue().getExtraFieldExists(findex))
										series.add(entry.getKey(),m*entry.getValue().getExtraFields(findex)+c);
								}
							}
									
							//if (!isTemperature(listfields.get(f)))
							//dataset.addSeries(axis,series);	
														
							if (series.getMinX() < minTime)
								minTime = (long)series.getMinX();
							if (series.getMaxX() > maxTime)
								maxTime = (long)series.getMaxX();
							
							dataset.addSeries(series);
							
							//else
							//	dataset.addSeries(f,series);
							
							
							XYSeriesRenderer renderer = new XYSeriesRenderer();
							renderer.setColor(colour);
							renderer.setLineWidth(4);

							//stroke style
							mrend.addSeriesRenderer(renderer);
							
							cnt++;
						}
					}
				}
				float lasthsv[]= new float[3];
	    		Color.colorToHSV(colour,lasthsv);
	    		lasthsv[0] = (lasthsv[0] + (26)) % 360;
	    		colour = Color.HSVToColor(lasthsv);
			}		
			
		}
		
		if (cnt == 0)
			return null;

		//mrend.addXTextLabel(System.currentTimeMillis(), "custom label");
		int steps = 13;
		long inc = (maxTime-minTime)/steps;
		for (int i = 0; i < steps; i++)
		{
			Date dt = new Date(minTime + i*inc);
			SimpleDateFormat dtf = new SimpleDateFormat("HH:mm");

			mrend.addXTextLabel(minTime + i*inc, dtf.format(dt));
		}
		
		//mrend.setChartTitle("Altitude Plot");
		mrend.setYTitle("Altitude (m)",0);
		if (Math.min(2, differentFields) > 1){
			//mrend.addYTextLabel(10, "New Test", 1);
			//mrend.setYTitle("Hours", 1);
			mrend.setYAxisAlign(Align.RIGHT, 1);
		}
		
		mrend.setShowGrid(true);
		
		for (int i = 0 ; i < Math.min(2, differentFields); i++)
		{
			mrend.setXAxisMax(maxTime, i);
			mrend.setXAxisMin(minTime, i);
		}
			
		
		mrend.setYLabelsAngle(270);
		
		
		
		mrend.setXLabels(0);


		return ChartFactory.getLineChartView(context, dataset, mrend);
		
	}
	
}